python list comprehension if else